package com.cy;

import java.util.ArrayList;
import java.util.List;

public class GraphFactory {

    public Graph<InitSubStep> draw(){
        InitSubStep step1 = new InitSubStep("打开冰箱",1);
        InitSubStep step2 = new InitSubStep("放入大象",2);
        InitSubStep step3 = new InitSubStep("关上冰箱",3);
        List<InitSubStep> nodes = new ArrayList<>();
        nodes.add(step1);
        nodes.add(step2);
        nodes.add(step3);
        Graph<InitSubStep> graph = new Graph<>(nodes);
        graph.setChild(step1,step2);
        graph.setChild(step2,step3);
        return graph;
    }
}
